package com.hongyi.common.excel.core;

import cn.hutool.core.util.StrUtil;
import com.alibaba.excel.context.AnalysisContext;
import com.alibaba.excel.event.AnalysisEventListener;
import com.alibaba.excel.exception.ExcelAnalysisException;
import com.alibaba.excel.exception.ExcelDataConvertException;
import com.alibaba.fastjson2.JSON;
import com.hongyi.common.framework.utils.common.CollectionUtils;
import com.hongyi.common.framework.utils.string.StringUtils;
import com.hongyi.common.framework.utils.ValidatorUtils;
import lombok.NoArgsConstructor;
import lombok.extern.slf4j.Slf4j;


import javax.validation.ConstraintViolation;
import javax.validation.ConstraintViolationException;
import java.util.Map;
import java.util.Objects;
import java.util.Set;
import java.util.stream.Collectors;

/**
 * Excel 导入监听
 *
 * <p>
 * 此监听器继承自 {@link AnalysisEventListener}，并实现了 {@link ExcelListener} 接口。
 * </p>
 *
 * <p>
 * {@link AnalysisEventListener} 是 EasyExcel 提供的一个基础监听器类，用于处理 Excel 文件解析过程中的各种事件，
 * 包括每行数据解析完成后的回调、解析完成后的回调以及异常处理等。
 * </p>
 *
 * <p>
 * {@link ExcelListener} 是自定义的监听器,实现了ReadListener,主要目的是通过重写getExcelResult来获取读取结果。
 * </p>
 *
 * @author DingHao
 */
@Slf4j
@NoArgsConstructor
public class DefaultExcelListener<T> extends AnalysisEventListener<T> implements ExcelListener<T> {

    /**
     * 是否Validator检验，默认为是
     */
    private Boolean isValidate = Boolean.TRUE;

    /**
     * excel 表头数据
     */
    private Map<Integer, String> headMap;

    /**
     * 导入回执
     */
    private ExcelResult<T> excelResult;

    // 有个很重要的点 easyExcel的监听器不能被spring管理，要每次读取excel都要new,然后里面用到spring可以构造方法传进去
    public DefaultExcelListener(boolean isValidate) {
        this.excelResult = new DefaultExcelResult<>();
        this.isValidate = isValidate;
    }

    /**
     * 处理异常
     *
     * @param exception ExcelDataConvertException
     * @param context   Excel 上下文
     */
    @Override
    public void onException(Exception exception, AnalysisContext context) throws Exception {
        String errMsg = null;
        if (exception instanceof ExcelDataConvertException) {//数据格式解析异常,例如Integer类型的字段数据是String
            ExcelDataConvertException excelDataConvertException = (ExcelDataConvertException) exception;
            // 如果是某一个单元格的转换异常 能获取到具体行号
            Integer rowIndex = excelDataConvertException.getRowIndex();
            Integer columnIndex = excelDataConvertException.getColumnIndex();
            //字符串格式化拼接,相当于log.info的拼接方式
            errMsg = StrUtil.format("第{}行-第{}列-表头{}: 数据格式解析异常<br/>",
                rowIndex + 1, columnIndex + 1, headMap.get(columnIndex));
            if (log.isDebugEnabled()) {
                log.error(errMsg);
            }
        }
        if (exception instanceof ConstraintViolationException ) {//校验异常返回结果
            ConstraintViolationException constraintViolationException = (ConstraintViolationException) exception;
            // 获取违反约束的信息集合
            Set<ConstraintViolation<?>> constraintViolations = constraintViolationException.getConstraintViolations();
            //String constraintViolationsMsg = StreamUtils.join(constraintViolations, ConstraintViolation::getMessage, ", ");
            //将违反约束的信息用","分隔连接成一个字符串constraintViolationsMsg
            String constraintViolationsMsg;
            if (CollectionUtils.isEmpty(constraintViolations)) {
                constraintViolationsMsg = StringUtils.EMPTY;
            }else {
                constraintViolationsMsg = constraintViolations.stream()
                        .map(ConstraintViolation::getMessage)
                        .filter(Objects::nonNull)
                        .collect(Collectors.joining(", "));
            }
            //字符串格式化拼接,相当于log.info的拼接方式
            errMsg = StrUtil.format("第{}行数据校验异常: {}", context.readRowHolder().getRowIndex() + 1, constraintViolationsMsg);
            if (log.isDebugEnabled()) {
                log.error(errMsg);
            }
        }
        //将异常信息存入错误列表中
        excelResult.getErrorList().add(errMsg);
        throw new ExcelAnalysisException(errMsg);
    }

    /**
     * 处理 xcel表头信息的方法
     *
     * @param headMap 表头数据
     * @param context EasyExcel 提供的一个上下文对象，它包含了当前解析过程中的各种信息
     */
    @Override
    public void invokeHeadMap(Map<Integer, String> headMap, AnalysisContext context) {
        this.headMap = headMap;
        log.debug("解析到一条表头数据: {}", JSON.toJSONString(headMap));
    }

    /**
     * 每读取一行时执行的方法
     *
     * @param data 读取行的具体数据
     * @param context EasyExcel 提供的一个上下文对象，它包含了当前解析过程中的各种信息
     */
    @Override
    public void invoke(T data, AnalysisContext context) {
        //解析每一行数据会进行参数校验
        if (isValidate) {
            ValidatorUtils.validate(data);
        }
        excelResult.getList().add(data);//将读取的行数据存到回执数据的对象列表
    }

    /**
     * 数据解析完成后调用的方法
     *
     * @param context EasyExcel 提供的一个上下文对象，它包含了当前解析过程中的各种信息
     */
    @Override
    public void doAfterAllAnalysed(AnalysisContext context) {
        log.debug("所有数据解析完成！");
    }

    /**
     * 获取Excel读取结果
     */
    @Override
    public ExcelResult<T> getExcelResult() {
        return excelResult;
    }

}
